package leetcode101.greedy_strategy;

import java.util.Map;
import java.util.TreeMap;

/**
 * @author Synhard
 * @version 1.0
 * @Class Code3
 * @Description 给定一个区间的集合，找到需要移除区间的最小数量，使剩余区间互不重叠。
 * 注意:
 *
 * 可以认为区间的终点总是大于它的起点。
 * 区间 [1,2] 和 [2,3] 的边界相互“接触”，但没有相互重叠。
 * 示例 1:
 *
 * 输入: [ [1,2], [2,3], [3,4], [1,3] ]
 *
 * 输出: 1
 *
 * 解释: 移除 [1,3] 后，剩下的区间没有重叠。
 * 示例 2:
 *
 * 输入: [ [1,2], [1,2], [1,2] ]
 *
 * 输出: 2
 *
 * 解释: 你需要移除两个 [1,2] 来使剩下的区间没有重叠。
 * 示例 3:
 *
 * 输入: [ [1,2], [2,3] ]
 *
 * 输出: 0
 *
 * 解释: 你不需要移除任何区间，因为它们已经是无重叠的了。
 * @tel 13001321080
 * @email 823436512@qq.com
 * @date 2021-03-22 9:43
 */
public class Code3 {

    public static void main(String[] args) {
        int[][] arr = new int[][] {{1,2},{2,3},{3,4},{-100,-2},{5,7}};
        eraseOverlapIntervals(arr);
    }

    public static int eraseOverlapIntervals(int[][] intervals) {
        // 1.首先按照每一个数组的起始区间从小到大进行排序
        quickSort(intervals, 0, intervals.length - 1);
        TreeMap<Integer, Integer> removeBigger = new TreeMap<>();

        /*
         2. 对于区间开头相同若干区间我们只取结尾最小的那个，将结果保存在TreeMap中，因为TreeMap是根据添加顺序排序的所以应该用TreeMap
         不能使用HashMap
         */
        for (int[] interval : intervals) {
            Integer already = removeBigger.get(interval[0]);
            if (already == null || already > interval[1]) {
                removeBigger.put(interval[0], interval[1]);
            }
        }

        // 3. 记录去除了几个大结尾的区间
        int diff = intervals.length - removeBigger.size();

        int[][] retainSmaller = new int[removeBigger.size()][2];
        int rSPointer = 0;
        // 4. 对剩余的区间进行判断，如果相邻区间中前区间的结尾大于后区间的开头则保留结尾小的区间
        for (Map.Entry entry : removeBigger.entrySet()) {
            int key = (int) entry.getKey();
            int val = (int) entry.getValue();
            // 如果数组为空的话直接压入
            if (rSPointer == 0) {
                retainSmaller[rSPointer++] = new int[]{key, val};
            }
            // 如果数组非空的话进行比较操作
            // 如果前一个的结尾大于后一个的开头则选择保留结尾较小的
            else if (retainSmaller[rSPointer - 1][1] > key) {
                if (retainSmaller[rSPointer - 1][1] > val) {
                    retainSmaller[rSPointer - 1] = new int[]{key, val};
                }
                diff++;
            } else {
                retainSmaller[rSPointer++] = new int[]{key, val};
            }
        }
        return diff;
    }

    // 快速排序相关，用来根据初始位置进行排序
    public static void quickSort(int[][] arr, int low, int high) {
        if (low < high) {
            int pivot = partition(arr, low, high);
            quickSort(arr, low, pivot - 1);
            quickSort(arr, pivot + 1, high);
        }
    }

    public static int partition(int[][] arr, int low, int high) {
        int[] temp = arr[low];
        while (low < high) {
            while (low < high && arr[high][0] >= temp[0]) {
                high--;
            }
            arr[low] = arr[high];
            arr[high] = temp;
            while (low < high && arr[low][0] <= temp[0]) {
                low++;
            }
            arr[high] = arr[low];
            arr[low] = temp;
        }
        return low;
    }
}
